-
Notifications
You must be signed in to change notification settings - Fork 70
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Migrate Brain Image Synthesis Latent Diffusion from GenerativeModels to model-zoo #722
base: dev
Are you sure you want to change the base?
Conversation
…s repo. Signed-off-by: Virginia <[email protected]>
… that no longer applies.
for more information, see https://pre-commit.ci
… longer necessary. On inference, modification of the entry point from save_jpg to run to allow for run to be called alone, and truncation of loading statements which are too long. Modification of the README to include command example changes.
…z/model-zoo into xray_classifier_port
for more information, see https://pre-commit.ci
For metadata, adjusted the versions, setting current to Initial Release. I also removed nibabel from the optional package requirements as this CXR produces JPG and does not deal with nifti files.
…z/model-zoo into xray_classifier_port
…erify_bundle to pass the check for model.pt, since the requirement for two models (autoencoder and diffusion_model) makes sense for them to keep their specific names. Modification of inference.json to add dummy attributes to pass the ConfigWorkflow check. Modification of large_files.yml so that models are .pt and not .pth.
…erify_bundle to pass the check for model.pt, since the requirement for two models (autoencoder and diffusion_model) makes sense for them to keep their specific names. Modification of inference.json to add dummy attributes to pass the ConfigWorkflow check. Modification of large_files.yml so that models are .pt and not .pth.
for more information, see https://pre-commit.ci
Add LICENSE file.
Modification of diffusion_model.pt name to model.pt to go through the verify_bundle Python function.
/build |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thanks for the contribution!
Leave several comments inline.
models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
Outdated
Show resolved
Hide resolved
models/brain_image_synthesis_latent_diffusion_model/docs/README.md
Outdated
Show resolved
Hide resolved
models/brain_image_synthesis_latent_diffusion_model/configs/metadata.json
Outdated
Show resolved
Hide resolved
models/brain_image_synthesis_latent_diffusion_model/configs/inference.json
Outdated
Show resolved
Hide resolved
models/brain_image_synthesis_latent_diffusion_model/scripts/saver.py
Outdated
Show resolved
Hide resolved
? |
/build |
FYI @garciadias, error from blossom.
|
/build |
Hi @garciadias , "/build" is used to trigger the CI pipeline, this PR looks good to me, and if all tests passed, I will merge it. Thanks! |
Hi @yiheng-wang-nv, thank you for explaining and reviewing this. I believe I fixed all the issues pointed out in the CI checks. I noticed the Many thanks! |
models/brain_image_synthesis_latent_diffusion_model/configs/metadata.json
Outdated
Show resolved
Hide resolved
…tadata.json Co-authored-by: Eric Kerfoot <[email protected]> Signed-off-by: Rafael Garcia-Dias <[email protected]>
Fixes # No issue assigned
Description
This is part of the effort on deprecating GenerativeModels and having all models available in the model-zoo instead.
Status
Ready
Please ensure all the checkboxes:
./runtests.sh --codeformat
.version
andchangelog
inmetadata.json
if changing an existing bundle.CONTRIBUTING.md
).monai
,pytorch
andnumpy
are correct inmetadata.json
.eval_metrics
of the provided weights and TorchScript modules.large_file.yml
./home/your_name/
for"bundle_root"
).